
from mcp.server.fastmcp import FastMCP

#定义一个MCP服务器
mcp = FastMCP("MCP Server")

@mcp.tool(name="add_tool",description="两个整数相加")
def add_tool(x:int,y:int):
    """
    这个函数用于计算两个整数的和
    Args:
        x (int): 第一个整数
        y (int): 第二个整数
    Returns:
        int: 两个整数的和
    """
    return x+y

@mcp.tool()
def sub_tool(x:int,y:int):
    """
    这个函数用于计算两个整数的差
    Args:
        x (int): 第一个整数
        y (int): 第二个整数
    Returns:
        int: 两个整数的差
    """
    return x-y


if __name__ == "__main__":
    #声明使用stdio传输协议,默认是sse协议
    # mcp.run(transport="sse")
    mcp.run(transport="stdio")